﻿using System;
using System.Collections.Generic;
using UnityEngine;
using System.Runtime.InteropServices;

namespace Obi{

	public class ObiDistanceFieldShapeTracker : ObiShapeTracker
	{
		public ObiDistanceField distanceField;
        ObiDistanceFieldHandle handle;


		public ObiDistanceFieldShapeTracker(ObiCollider source, Component collider, ObiDistanceField distanceField){

            this.source = source;
            this.collider = collider;
            this.distanceField = distanceField;
		}

        /**
		 * Forces the tracker to update distance field data during the next call to UpdateIfNeeded().
		 */
        public void UpdateDistanceFieldData()
        {
            ObiColliderWorld.GetInstance().DestroyDistanceField(handle);
        }

        public override void UpdateIfNeeded ()
        {

            bool trigger = false;
            if (collider is Collider) trigger = ((Collider)collider).isTrigger;
            else if (collider is Collider2D) trigger = ((Collider2D)collider).isTrigger;

            // retrieve collision world and index:
            var world = ObiColliderWorld.GetInstance();
            int index = source.Handle.index;

            // decrease reference count of current handle if the df data it points to is different
            // than the df used by the collider:
            if (handle != null && handle.owner != distanceField)
            {
                if (handle.Dereference())
                    world.DestroyDistanceField(handle);
            }

            if (handle == null || !handle.isValid)
            {
                handle = world.GetOrCreateDistanceField(distanceField);
                handle.Reference();
            }

            // update collider:
            var shape = world.colliderShapes[index];
            shape.type = ColliderShape.ShapeType.SignedDistanceField;
            shape.filter = source.Filter;
            shape.SetSign(source.Inverted);
            shape.isTrigger = trigger;
            shape.rigidbodyIndex = source.Rigidbody != null ? source.Rigidbody.Handle.index : -1;
            shape.materialIndex = source.CollisionMaterial != null ? source.CollisionMaterial.handle.index : -1;
            shape.forceZoneIndex = source.ForceZone != null ? source.ForceZone.Handle.index : -1;
            shape.contactOffset = source.Thickness;
            shape.dataIndex = handle.index;
            world.colliderShapes[index] = shape;

            // update bounds:
            var aabb = world.colliderAabbs[index];
            aabb.FromBounds(distanceField.FieldBounds.Transform(source.transform.localToWorldMatrix), shape.contactOffset);
            world.colliderAabbs[index] = aabb;

            // update transform:
            var trfm = world.colliderTransforms[index];
            trfm.FromTransform3D(source.transform, source.Rigidbody as ObiRigidbody);
            world.colliderTransforms[index] = trfm;
        }

        public override void Destroy()
        {
            base.Destroy();

            if (handle != null && handle.Dereference())
                ObiColliderWorld.GetInstance().DestroyDistanceField(handle);
        }

    }
}

